"""
    准备工作:
        pip install opencv-python
        pip install matplotlib
"""
import numpy as np
import cv2

# 彩色图像和灰度图像在数据结构上的主要区别在于颜色通道的数量
# 创建一个3x3像素的图像数据，每个像素包含RGB三个通道的值 (h, w, c)
color_image_data = [
    # 高度
    [
        # 宽度
        [166, 145, 153], [166, 145, 153], [166, 145, 154]  # 颜色通道(3) [B, G, R]
    ],
    [
        [159, 138, 147], [159, 138, 147], [157, 138, 147]
    ],
    [
        [159, 138, 147], [157, 138, 147], [156, 137, 146]
    ]
]

gray_image_data = [
    # 高度
    [
        # 宽度
        150, 150, 150  # 灰度值
    ],
    [
        143, 143, 143
    ],
    [
        143, 142, 142
    ]
]

# 将图像数据转换为uint8类型，范围为0-255
image_rgb_data = np.array(color_image_data).astype(np.uint8)
image_gray_data = np.array(gray_image_data).astype(np.uint8)

# 保存图像
cv2.imwrite('demo_rgb.jpg', image_rgb_data)
cv2.imwrite('demo_gray.jpg', image_gray_data)

